﻿/* -*- c++ -*- */
/*
 * Gqrx SDR: Software defined radio receiver powered by GNU Radio and Qt
 *           http://gqrx.dk/
 *
 * Copyright 2013 Christian Lindner DL2VCL, Stefano Leucci.
 *
 * Gqrx is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3, or (at your option)
 * any later version.
 *
 * Gqrx is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Gqrx; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street,
 * Boston, MA 02110-1301, USA.
 */
#include <QFile>
#include <QStringList>
#include "bookmarks.h"
#include "bookmarkstablemodel.h"


BookmarksTableModel::BookmarksTableModel(QObject *parent) :
	QAbstractTableModel(parent)
{
}

int BookmarksTableModel::rowCount ( const QModelIndex & /*parent*/ ) const
{
	return m_Bookmarks.size();
}
int BookmarksTableModel::columnCount ( const QModelIndex & /*parent*/ ) const
{
	return 5;
}

QVariant BookmarksTableModel::headerData ( int section, Qt::Orientation orientation, int role ) const
{
	if(orientation == Qt::Horizontal && role == Qt::DisplayRole)
	{
		switch(section)
		{
		case COL_FREQUENCY:
			return QString("Frequency");
			break;
		case COL_NAME:
			return QString("Name");
			break;
		case COL_MODULATION:
			return QString("Modulation");
			break;
		case COL_BANDWIDTH:
			return QString("Bandwidth");
			break;
		case COL_TAGS:
			return QString("Tag");
			break;
		}
	}
	if(orientation == Qt::Vertical && role == Qt::DisplayRole)
	{
		return section;
	}
	return QVariant();
}

QVariant BookmarksTableModel::data ( const QModelIndex & index, int role ) const
{
	BookmarkInfo& info = *m_Bookmarks[index.row()];

	if(role==Qt::BackgroundRole)
	{
		QColor bg(info.GetColor());
		bg.setAlpha(0x60);
		return bg;
	}
	else if(role == Qt::DisplayRole || role==Qt::EditRole)
	{
		switch(index.column())
		{
		case COL_FREQUENCY:
				return info.frequency;
		case COL_NAME:
				return (role==Qt::EditRole)?QString(info.name):info.name;
		case COL_MODULATION:
				return info.modulation;
		case COL_BANDWIDTH:
			return (info.bandwidth==0)?QVariant(""):QVariant(info.bandwidth);
		 case COL_TAGS:
			QString strTags;
			for(int iTag=0; iTag<info.tags.size(); ++iTag)
			{
				if(iTag!=0)
				{
					strTags.append(",");
				}
				TagInfo& tag = *info.tags[iTag];
				strTags.append(tag.name);
			}
			return strTags;
		}
	}
	return QVariant();
}

bool BookmarksTableModel::setData(const QModelIndex &index, const QVariant &value, int role)
{
	if(role==Qt::EditRole)
	{
		BookmarkInfo &info = *m_Bookmarks[index.row()];
		switch(index.column())
		{
		case COL_FREQUENCY:
			{
				info.frequency = value.toLongLong();
				emit dataChanged(index, index);
			}
			break;
		case COL_NAME:
			{
				info.name = value.toString();
				emit dataChanged(index, index);
				return true;
			}
			break;
		case COL_MODULATION:

			break;
		case COL_BANDWIDTH:
			{
				info.bandwidth = value.toInt();
				emit dataChanged(index, index);
			}
			break;
		case COL_TAGS:
			{
				info.tags.clear();
				QString strValue = value.toString();
				QStringList strList = strValue.split(",");
				for(int i=0; i<strList.size(); ++i)
				{
					QString strTag = strList[i].trimmed();
					info.tags.append( &Bookmarks::Get().findOrAddTag(strTag) );
				}
				emit dataChanged(index, index);
				return true;
			}
			break;
		}
		return true; // return true means success
	}
	return false;
}

Qt::ItemFlags BookmarksTableModel::flags ( const QModelIndex& index ) const
{
	Qt::ItemFlags flags = Qt::NoItemFlags;

	switch(index.column())
	{
	case COL_FREQUENCY:
	case COL_NAME:
	case COL_BANDWIDTH:
	case COL_MODULATION:
		flags = Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsEditable;
		break;
	case COL_TAGS:
		flags = Qt::ItemIsEnabled | Qt::ItemIsSelectable;
		break;
	}
	return flags;
}

void BookmarksTableModel::update()
{
	int iRow = 0;
	m_Bookmarks.clear();
	for(int iBookmark=0; iBookmark<Bookmarks::Get().size(); iBookmark++)
	{
		BookmarkInfo& info = Bookmarks::Get().getBookmark(iBookmark);

		bool bActive = false;
		for(int iTag=0; iTag<info.tags.size(); ++iTag)
		{
			TagInfo& tag = *info.tags[iTag];
			if(tag.active)
			{
				bActive = true;
				break;
			}
		}
		if(bActive)
		{
			m_mapRowToBookmarksIndex[iRow]=iBookmark;
			m_Bookmarks.append(&info);
			++iRow;
		}
	}

	emit layoutChanged();
}

BookmarkInfo *BookmarksTableModel::getBookmarkAtRow(int row)
{
	return m_Bookmarks[row];
}

int BookmarksTableModel::GetBookmarksIndexForRow(int iRow)
{
  return m_mapRowToBookmarksIndex[iRow];
}

